#-*- coding:utf-8 -*-
from setuptools import setup
from torch.utils.cpp_extension import CppExtension, CUDAExtension, BuildExtension

setup(name='gdwconv',
      ext_modules=[CUDAExtension('gdwconv', ['gdw_conv_cuda.cpp', 'gdw_conv_cuda_kernel.cu'])],
      cmdclass={'build_ext': BuildExtension})